source('../../Data/get_dermatologist_data.R')
DF = get_dermatologist_data()

mr_range = range(DF$Medicare_reimbursement)
et_range = range(DF$Estimated_time)

mask = DF$Medical_or_Cosmetic == 'M'
df_med = DF[mask, ]
lm_med = lm(Medicare_reimbursement ~ Estimated_time, data=df_med)

mask = DF$Medical_or_Cosmetic == 'C'
df_cos = DF[mask, ]
lm_cos = lm(Medicare_reimbursement ~ Estimated_time, data=df_cos)

plot(df_med$Estimated_time, df_med$Medicare_reimbursement, xlim=et_range, ylim=mr_range, type='p', pch=19, col='blue', xlab='Time', ylab='Cost')
abline(lm_med, col='blue')
lines(df_cos$Estimated_time, df_cos$Medicare_reimbursement, type='p', pch=19, col='red')
abline(lm_cos, col='red')
legend('topleft', c('medical', 'cosmetic'), col=c('blue', 'red'), lty=1)
grid()